Rethinking Routing Exploration in Modern Mixture-of-Experts
重新思考现代 MoE 路由探索:MirrorMoE
🧭 MoE 预训练路由探索优化 ★ 深度解读
ETH Zurich · The University of Hong Kong · Max Planck Institute for Intelligent Systems · WU Vienna
MirrorMoE 通过 token 自适应抑制低概率路由专家缩小探索空间,解决细粒度共享专家 MoE 的探索困境并稳定涨点
📌 工作参考价值(4 点)
- 若服务链路使用 DeepSeek/Qwen3 等 MoE 基座做 embedding、生成式推荐或 Agent 策略,当启用或调大 fine-grained top-k 时,不要假设 shared experts 一定带来收益;可先做单 expert 消融或随机路由诊断,确认 shared experts 是否主导输出并压制路由探索。
- 训练阶段可采用 MirrorMoE 式“选择分布与聚合分布分离”:用镜像 logits 选专家,用原始 logits 权重聚合,再把 rho 从 0.4 cosine 退火到 0;工程实现只需额外一次 softmax 和符号翻转,训练成本几乎不增。
- 负载均衡 loss 应加在实际贡献概率 xi 上,而不是选择概率 p 上;当低权重专家被激活但几乎不贡献时,这种设计能避免均衡目标误导路由,可迁移到多兴趣/多任务 MoE 推荐模型。
- 若模型不含 shared experts,MirrorMoE 提升有限;不要盲目上该 trick,应优先用于 shared-expert + 较大 K 的配置,并在增大 K 时重点观察是否出现“K 涨指标不涨甚至下降”的探索困境。